package com.jiuye.controller;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.jiuye.common.GlobalConstant;
import com.jiuye.common.R;
import com.jiuye.entity.User;
import com.jiuye.service.IUserService;
import com.jiuye.utils.ValidateCodeUtils;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiImplicitParams;
import io.swagger.annotations.ApiOperation;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import javax.servlet.http.HttpSession;
import java.util.Map;
import java.util.concurrent.TimeUnit;

/**
 * 发送短信验证码表示层
 *
 * @author likeWen
 * @since 2022/11/5 21:39
 */
@Slf4j
@RestController
@RequestMapping("/user")
@Api(tags = "用户登录接口")
public class UserController {
    @Autowired
    private IUserService userService;

    @Autowired
    private RedisTemplate<Object, Object> redisTemplate;

    /**
     * 发送短信
     */
    @PostMapping("/sendMsg")
    @ApiOperation(value = "发送短信接口")
    public R<String> sendMsg(@RequestBody User user, HttpSession session) {
        //1.获取用户手机号码
        String phone = user.getPhone();
        //2.生成随机数
        String code = ValidateCodeUtils.generateValidateCode(4).toString();
        //3.调用短信工具类发送短信
        //SMSUtils.sendMessage("阿里云短信测试", "SMS_154950909", phone, code);
        log.info("code---->{}", code);
        //4.写入到session中
        // session.setAttribute(phone, code);
        //写入到redis中，并设置一个5分钟的有效期
        //第一个参数，key
        //第二个参数，value
        //第三个参数，时间
        //第四个参数，时间单位
        this.redisTemplate.opsForValue().set(phone, code, 5, TimeUnit.MINUTES);
        return R.success(GlobalConstant.SUCCESS);
    }

    /**
     * 手机登录
     */
    @PostMapping("/login")
    @ApiOperation(value = "验证码校验接口")
    public R<User> login(@RequestBody Map map, HttpSession session) {
        //1.获取前端传递的手机号和验证码
        //手机号
        String phone = map.get("phone").toString();
        //前端传入的验证码
        String newCode = map.get("code").toString();
        //2.从session中获取手机号中对应的正确的验证码
        //正确的验证码
        //String oldCode = (String) session.getAttribute(phone);
        //2.从redis中获取手机号中对应的正确的验证码
        String oldCode = (String) this.redisTemplate.opsForValue().get(phone);
        /**
         * 为了更方便的进行代码校验，设置一个万能ID
         */
        if ("15732243291".equals(phone) && "0".equals(newCode)) {
            oldCode = "0";
        }
        //3.进行验证码对比，如果不一致，直接返回登录失败
        if (!newCode.equals(oldCode)) {
            return R.error("验证码错误，登录失败");
        }
        // 4). 如果比对成功，需要根据手机号查询当前用户，
        // 如果用不存在，则新建一个该手机号的用户。即注册一个新用户
        // 用手机号查询这个手机号,在用户表中是否存在
        // 如果存在就代表这个用户已经是我们的用户了,直接返回就可以
        // 如果不存在,就需要用这个用的手机号信息,新增一条记录.
        LambdaQueryWrapper<User> queryWrapper = new LambdaQueryWrapper<>();
        queryWrapper.eq(User::getPhone, phone);
        User user = this.userService.getOne(queryWrapper);
        //如果为空，就代表用户不存在，直接注册
        if (user == null) {
            user = new User();
            user.setPhone(phone);
            // 用户状态
            // 1是启用
            user.setStatus(1);
            this.userService.save(user);
        }
        // 5). 将登录用户的ID存储在Session中
        session.setAttribute(GlobalConstant.MOBILE_ID, user.getId());
        //删除redis中的短信验证码
        this.redisTemplate.delete(phone);
        return R.success(user);
    }
}
